基于YOLOv8+Pyqt5火焰烟雾检测系统 您所在的位置:网站首页 pyqt yolo训练 基于YOLOv8+Pyqt5火焰烟雾检测系统

基于YOLOv8+Pyqt5火焰烟雾检测系统

2024-07-16 04:56| 来源: 网络整理| 查看: 265

1、YOLOv8的基本原理

YOLOv8是一种前沿的目标检测技术,它基于先前YOLO版本在目标检测任务上的成功,进一步提升了性能和灵活性。主要的创新点包括一个新的骨干网络、一个新的 Ancher-Free 检测头和一个新的损失函数,可以在从 CPU 到 GPU 的各种硬件平台上运行。其主要网络结构如下: 在这里插入图片描述

2、软件主要功能 可进行火焰、烟雾及正常这3种状态的目标检测; 支持图片、视频及摄像头进行检测,同时支持图片的批量检测; 界面可实时显示目标位置、目标总数、置信度、用时等信息; 支持图片或者视频的检测结果保存; 3、数据集与训练

通过网络上搜集关于火焰及烟雾的各类图片,并使用LabelMe标注工具对每张图片中的目标边框(Bounding Box)及类别进行标注。一共包含979张图片,其中训练集包含877张图片,验证集包含47张图片,测试集包含55张图片部分图像及标注。 在这里插入图片描述 图片数据的存放格式如下,在项目目录中新建datasets目录,同时将跌倒检测的图片分为训练集与验证集放入helmetData目录下。 在这里插入图片描述 同时我们需要新建一个data.yaml文件,用于存储训练数据的路径及模型需要进行检测的类别。YOLOv8在进行模型训练时,会读取该文件的信息,用于进行模型的训练与验证。data.yaml的具体内容如下:

train: E:\MyCVProgram\FireSmokeDetection\datasets\FireSmokeData\train # train images (relative to 'path') 128 images val: E


【本文地址】

公司简介

联系我们

今日新闻

    推荐新闻

    专题文章
      CopyRight 2018-2019 实验室设备网 版权所有